3272e413a0af438406aba74bb3fa4e3d10a4f515,h2o-algos/src/test/java/hex/word2vec/Word2VecTest.java,Word2VecTest,testW2V_SG_NS,#,122
Before Change
job = new Word2Vec(parms);
job.train();
w2vm = job.get();
printResults(w2vm.findSynonyms("dog",5));
} finally {
if( fr != null ) fr .remove();
if( job != null) job.remove();
After Change
job = new Word2Vec(parms);
job.train();
w2vm = job.get();
HashMap hm = w2vm.findSynonyms("dog",10);
printResults(hm);
Assert.assertTrue(hm.containsKey("dogs"));
} finally {
if( fr != null ) fr .remove();